import pandas as pd
import sqlite3
from openpyxl import load_workbook

# 读取Excel文件
df = pd.read_excel(r'D:\all_project\Trae项目\SCUM\output_1.xlsx', header=None, names=['original_code', 'chinese_name', 'game_code'], engine='openpyxl')

# 连接数据库
conn = sqlite3.connect('data.db')

# 插入数据时忽略重复项
conn.execute('DROP TABLE IF EXISTS codes')
conn.execute('''CREATE TABLE codes
                     (id INTEGER PRIMARY KEY AUTOINCREMENT,
                     original_code TEXT,
                     chinese_name TEXT,
                     game_code TEXT,
                     search_count INTEGER DEFAULT 0)''')
df.to_sql('codes', conn, if_exists='append', index=False, dtype={
    'chinese_name': 'TEXT',
    'original_code': 'TEXT',
    'game_code': 'TEXT'
})

print('数据导入完成，共导入', len(df), '条记录')
conn.close()